package com.haidechizi.sort;

import java.util.Arrays;
import java.util.LinkedList;
import java.util.Queue;

/**
 * @author: wangdejun
 * @create: 2020-03-14 10:58
 **/
public class RadixSort {

    public static void main(String[] args) {
        int[] arr = new int[]{81, 3223, 55, 10, 1, 478, 66, 73};
        radixSortByQueue(arr);
        System.out.println(Arrays.toString(arr));

    }

    /**
     * 基数排序
     * 循环比较数组中最大数的位数轮
     * 将每轮中相应位数的数相同的数放入到相应的桶中
     * 将数按照桶中的顺序依次取出，放入到原数组中
     *
     * @param arr
     */
    public static void radixSort(int[] arr) {
        int max = arr[0];
        for (int i = 1; i < arr.length; i++) {
            if (max < arr[i]) {
                max = arr[i];
            }
        }

        // 获取最大数的位数
        int maxLength = String.valueOf(max).length();
        // 定义临时数组(尾数只有十个，每个中最多将整个arr存进去)
        int[][] temp = new int[10][arr.length];

        // 记录每个桶中元素的个数
        int[] counts = new int[10];

        for (int i = 0, n = 1; i < maxLength; i++, n *= 10) {
            // 循环比较数组中的每一个数
            for (int j = 0; j < arr.length; j++) {
                // 获取相应位上的数
                int ys = arr[j] / n % 10;
                temp[ys][counts[ys]] = arr[j];
                counts[ys]++;
            }
            int index = 0;
            // 将临时数组中的数放入到原数组中
            for (int k = 0; k < counts.length; k++) {
                if (counts[k] > 0) {
                    for (int l = 0; l < counts[k]; l++) {
                        arr[index] = temp[k][l];
                        index++;
                    }
                    // 清空临时数组每个节点记录的元素个数
                    counts[k] = 0;
                }
            }

        }


    }

    public static void radixSortByQueue(int[] arr) {
        int max = arr[0];
        for (int i = 1; i < arr.length; i++) {
            if (max < arr[i]) {
                max = arr[i];
            }
        }

        int maxLength = String.valueOf(max).length();
        //
        Queue<Integer>[] temp = new Queue[10];
        for (int i = 0, n = 1; i < maxLength; i++, n *= 10) {
            for (int j = 0; j < arr.length; j++) {
                int ys = arr[j] / n % 10;
                Queue<Integer> queue = temp[ys];
                if (queue == null) {
                    queue = new LinkedList<Integer>();
                }
                queue.add(arr[j]);
                temp[ys] = queue;
            }
            // 将队列的元素取出
            int index = 0;
            for (Queue<Integer> queue : temp) {
                while (queue != null && !queue.isEmpty()) {
                    int data = queue.poll();
                    arr[index] = data;
                    index++;
                }
            }
        }

    }
}
